//
//  Problem791.swift
//  TestProject
//
//  Created by 武侠 on 2021/3/24.
//  Copyright © 2021 zhulong. All rights reserved.
//

import UIKit

/*
 791. 自定义字符串排序
 字符串S和 T 只包含小写字符。在S中，所有字符只会出现一次。
 S 已经根据某种规则进行了排序。我们要根据S中的字符顺序对T进行排序。更具体地说，如果S中x在y之前出现，那么返回的字符串中x也应出现在y之前。
 返回任意一种符合条件的字符串T。
 示例:
     输入:
     S = "cba"
     T = "abcd"
     输出: "cbad"
     解释:
     S中出现了字符 "a", "b", "c", 所以 "a", "b", "c" 的顺序应该是 "c", "b", "a".
     由于 "d" 没有在S中出现, 它可以放在T的任意位置. "dcba", "cdba", "cbda" 都是合法的输出。
 注意:
     S的最大长度为26，其中没有重复的字符。
     T的最大长度为200。
     S和T只包含小写字符。
 */
@objcMembers class Problem791: NSObject {
    func solution() {
        print(customSortString("cba", "abcd"))
    }
    
    /*
     题目要求：按照S的字符顺序排序。那么S的字符顺序就是字符的下标
     "cba" => (0,1,2) => c=>0 b=>1 a=>2
     那么排序T时，先获取字符对应的下标值，没有的话，取最大，然后对比排序
     */
    func customSortString(_ S: String, _ T: String) -> String {
        var dic: [Character: Int] = [:]
        for (i, c) in S.enumerated() {
            dic[c] = i
        }
        
        print(dic)
        let array = T.sorted { (c1, c2) -> Bool in
            let ci1 = dic[c1] ?? S.count
            let ci2 = dic[c2] ?? S.count
            return ci1 < ci2
        }
        return array.reduce("") { (c1, c2) -> String in
            return String(c1) + String(c2)
        }
    }
}
